package com.zhugang.week10;

/**
 * @program algorithms
 * @description: canPartition
 * @author: chanzhugang
 * @create: 2022/08/28 16:18
 */
public class CanPartition {

    /**
     * 416 分割等和子集
     *
     * @param nums
     * @return
     */
    public boolean canPartition(int[] nums) {
        // 把数组分割成两个子集，子集元素之和是否能相等
        // 抽象成0-1背包问题，是否能装满背包
        int n = nums.length;
        int sum = 0;
        for (int i = 0; i < n; i++) {
            sum += nums[i];
        }
        if (sum % 2 == 1) {
            // 和为奇数
            return false;
        }
        sum = sum / 2;
        boolean[][] dp = new boolean[n][sum + 1];
        dp[0][0] = true;
        // 初始化第一行
        if (nums[0] <= sum) {
            dp[0][nums[0]] = true;
        }
        for (int i = 1; i < n; i++) {
            for (int j = 0; j <= sum; j++) {
                if (j - nums[i] >= 0) {
                    // 没越界：装或者不装
                    dp[i][j] = dp[i - 1][j] || dp[i - 1][j - nums[i]];
                } else {
                    // 越界：不装
                    dp[i][j] = dp[i - 1][j];
                }
            }
        }
        // 最后一行是否有满足条件
        return dp[n - 1][sum];
    }
}